import argparse

# import torch


def get_args_parser():
    parser = argparse.ArgumentParser(description='RL')
    parser.add_argument('--env-name',
                        type=str,
                        default='MontezumaRevengeNoFrameskip-v0')
    # default='MontezumaRevengeNoFrameskip-v0')
    parser.add_argument('--num-stacks', type=int, default=8)
    parser.add_argument('--num-steps', type=int, default=1000)
    parser.add_argument('--test-steps', type=int, default=2000)
    parser.add_argument('--instant-save-labels', type=bool, default=True)
    parser.add_argument('--num-frames', type=int, default=100000)

    ## other parameter
    parser.add_argument(
        '--log-interval',
        type=int,
        # default=10,
        default=1,
        help='log interval, one log per n updates (default: 10)')
    parser.add_argument('--save-img', type=bool, default=True)
    parser.add_argument(
        '--save-interval',
        type=int,
        default=10,
        help='save interval, one eval per n updates (default: None)')
    parser.add_argument('--play-game', type=bool, default=False)
    parser.add_argument(
        '-c',
        '--config_yaml',
        default=
        '/home/zhaoxiaoyu/courses/rl/assignments/RL_HW1_Dagger/Dagger/model/config/train.yml',
        type=str,
        metavar='FILE',
        help='YAML config file specifying default arguments')

    return parser